Update app.py

app.py

@@ -10,6 +10,9 @@ from huggingface_hub import HfApi, create_repo
 import importlib
 import os
 from huggingface_hub import HfApi, create_repo
+from transformers import pipeline
+codex_pipeline = pipeline("code-generation", model="EleutherAI/code-davinci-002")
+import huggingface_cli
 
 hf_api = HfApi()
 
@@ -288,17 +291,59 @@ def run_docker_container(image_name, port):
     container = client.containers.run(image_name, detach=True, ports={f'{port}/tcp': port})
     return container
 
-def deploy_space_to_hf(project_name, hf_token):
-    repository_name = f"my-awesome-space_{datetime.now().timestamp()}"
-    files = get_built_space_files()
-    commit_response = deploy_to_git(project_name, repository_name, files)
-    if commit_response:
-        api = HfApi(token=hf_token)  # Corrected import statement
-        api.create(repository_name, files=files, push_to_hub=True)
-        return f"Space '{repository_name}' deployed successfully."
-    else:
-        return "Failed to commit changes to Space."
+def generate_app(user_idea, project_name):
+    # Extract key information from the user idea
+    summary = nlp_pipeline(user_idea, max_length=50, min_length=10)[0]["summary_text"]
+
+    # Create project directory if it doesn't exist
+    project_path = create_project(project_name)
+
+    # Generate code using Codex
+    prompt = f"Create a simple Streamlit app for the project named '{project_name}'. The app should display the following summary: '{summary}'."
+    generated_code = codex_pipeline(prompt)[0]['generated_text']
+
+    # Save the generated code to a file in the project directory
+    with open(os.path.join(project_path, "app.py"), "w") as f:
+        f.write(generated_code)
+
+    # Deploy the app to Hugging Face Spaces
+    deploy_app_to_hf_spaces(project_name, token, generated_code)
+
+    return generated_code, project_path
+
+def deploy_app_to_hf_spaces(project_name, token, generated_code):
+    repo_name = f"hf-{project_name}"
+    repo_id = hf_api.changelog.get_repo_id(repo_name)
+
+    if not repo_id:
+        create_repo = huggingface_cli.create_repo(repo_name, "public", "Streamlit App", token)
+        repo_id = create_repo["repo_id"]
 
+    # Save the generated code to a temporary file
+
+    temp_file = "temp_code.py"
+    with open(temp_file, "w") as f:
+        f.write(generated_code)
+
+    # Upload the file to Hugging Face Spaces
+    api.upload_files(repo_id, [temp_file], token)
+
+    # Delete the temporary file
+    os.remove(temp_file)
+
+    def launch_chatapp(project_path):
+    # ... (previous code)
+
+    if st.button("Build & Deploy"):
+        token = st.text_input("Enter your Hugging Face token:", type="password")
+        generated_code, project_path = generate_app(st.session_state["user_idea"], project_name)
+        st.write(f"Generated code: {generated_code}")
+        if token:
+            deploy_app_to_hf_spaces(project_name, token, generated_code)
+            st.success("App deployed to Hugging Face Spaces!")
+        else:
+            st.error("Please enter a Hugging Face token to deploy the app.")
+    
 if __name__ == "__main__":
     db.create_all()  # Create the database tables if they don't exist
     main()