Upload app.py with huggingface_hub

app.py

@@ -1,11 +1,12 @@
 import streamlit as st
 from huggingface_hub import HfApi, create_repo, upload_file
-from groq import Groq
 import os
+import requests
+import json
 
-# Set up Groq API key
-groq_api_key = st.secrets["devassistapi"]
-client = Groq(api_key=groq_api_key)
+# Set up Google API key
+google_api_key = st.secrets["googleapi"]
+google_api_url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key={google_api_key}"
 
 # App title
 st.title("HF DevAssist - Create and Manage Hugging Face Spaces")
@@ -50,25 +51,33 @@ if st.button("Generate and Deploy App"):
         st.error("App description cannot be empty.")
     else:
         try:
-            # Generate app code using Groq
-            st.info("Generating app code using Groq API...")
-            chat_completion = client.chat.completions.create(
-                messages=[
+            # Generate app code using Google API
+            st.info("Generating app code using Google API...")
+            payload = {
+                "contents": [
                     {
-                        "role": "user",
-                        "content": (
-                            f"Generate a Streamlit app based on this description: "
-                            f"'{app_description}'"
-                        ),
+                        "parts": [{"text": f"Generate a Streamlit app with the following description: {app_description}. Include the app code followed by '### requirements.txt' and the dependencies required."}]
                     }
-                ],
-                model="llama-3.3-70b-versatile",
-            )
-            generated_code = chat_completion.choices[0].message.content
+                ]
+            }
+            headers = {"Content-Type": "application/json"}
+            response = requests.post(google_api_url, headers=headers, json=payload)
+
+            if response.status_code != 200:
+                st.error(f"Google API request failed with status code {response.status_code}: {response.text}")
+                st.stop()
 
-            # Split generated code into `app.py` and `requirements.txt`
-            app_code = generated_code.split("### requirements.txt")[0].strip()
-            req_code = generated_code.split("### requirements.txt")[1].strip()
+            generated_code = response.json()["contents"][0]["parts"][0]["text"]
+            st.text_area("Debug: Google API Response", generated_code, height=150)
+
+            # Split generated code
+            try:
+                app_code, req_code = generated_code.split("### requirements.txt")
+                app_code = app_code.strip()
+                req_code = req_code.strip()
+            except ValueError:
+                st.error("The Google API response did not contain the expected sections. Please try again.")
+                st.stop()
 
             # Create or update Hugging Face Space
             space_full_name = f"{user_info['name']}/{space_name}"