Update app.py

app.py

@@ -1,40 +1,32 @@
+import streamlit as st
 import requests
 import json
-import gradio as gr
 
-url="https://65ab-182-19-35-177.ngrok-free.app/api/generate"
+# Access the API key from Hugging Face Spaces secrets (assuming Streamlit Space)
+API_KEY = st.secrets["API_KEY"]  # Replace "API_KEY" with your actual secret key name
 
-headers={
-
-    'Content-Type':'application/json'
-}
-
-history=[]
+API_URL = "https://api-inference.huggingface.co/models/meta-llama/CodeLlama-7b-hf"
 
 def generate_response(prompt):
-    history.append(prompt)
-    final_prompt="\n".join(history)
-
-    data={
-        "model":"codingsensei",
-        "prompt":final_prompt,
-        "stream":False
-    }
-
-    response=requests.post(url,headers=headers,data=json.dumps(data))
-
-    if response.status_code==200:
-        response=response.text
-        data=json.loads(response)
-        actual_response=data['response']
-        return actual_response
-    else:
-        print("error:",response.text)
-
-
-interface=gr.Interface(
-    fn=generate_response,
-    inputs=gr.Textbox(lines=4,placeholder="Enter your Prompt"),
-    outputs="text"
-)
-interface.launch(share=True)
+  data = {
+    "inputs": prompt
+  }
+  headers = {"Authorization": f"Bearer {API_KEY}"}  # Use f-string for dynamic insertion
+  response = requests.post(API_URL, headers=headers, json=data)
+  if response.status_code == 200:
+    return response.json()
+  else:
+    st.error(f"Error: {response.status_code}")
+    st.error(response.text)
+    return None
+
+def main():
+  st.title("Code Assistant")
+  prompt = st.text_area("Enter your prompt", height=100)
+  if st.button("Submit"):
+    response = generate_response(prompt)
+    if response:
+      st.text_area("Response", value=json.dumps(response, indent=2), height=200)
+
+if __name__ == "__main__":
+  main()