Update app.py

app.py

@@ -5,6 +5,7 @@ from io import BytesIO
 from PIL import Image
 import numpy as np
 import traceback
+import json
 
 def generate_gradio_app(api_key, image):
     if not api_key:
@@ -40,7 +41,7 @@ def generate_gradio_app(api_key, image):
 Please generate the entire Gradio code based on the provided image."""
 
         # Make the API call
-        stream = client.chat.completions.create(
+        response = client.chat.completions.create(
             model="meta-llama/Llama-3.2-90B-Vision-Instruct-Turbo",
             messages=[
                 {
@@ -62,15 +63,17 @@ Please generate the entire Gradio code based on the provided image."""
             top_k=50,
             repetition_penalty=1,
             stop=["<|eot_id|>", "<|eom_id|>"],
-            stream=True
+            stream=False  # Changed to False for easier debugging
         )
         
-        # Collect the streamed response
-        generated_code = ""
-        for chunk in stream:
-            if chunk.choices[0].delta.content is not None:
-                generated_code += chunk.choices[0].delta.content
-                yield f"Generating Gradio code... (Current length: {len(generated_code)} characters)\n\n{generated_code}"
+        # Debug: Print the entire response
+        print("API Response:", json.dumps(response, indent=2))
+
+        # Extract the generated code from the response
+        if response.choices and response.choices[0].message:
+            generated_code = response.choices[0].message.content
+        else:
+            return "Error: Unexpected response structure from the API."
         
         if not generated_code:
             return "Error: No code generated from the model. Please try again."