Update app.py

app.py

@@ -63,7 +63,7 @@ def generate_software_spec(name, description, architecture, components, deployme
         max_tokens=2048,
     )
     print(response)
-    return response.choices[0].message["content"] if response and response.choices else "Error: No response received."
+    return response.choices[0].message.content if response and response.choices else "Error: No response received."
 
 # 🎨 Gradio UI
 with gr.Blocks() as demo:
@@ -84,8 +84,18 @@ with gr.Blocks() as demo:
     submit = gr.Button("Generate Design")
     output = gr.Markdown(label="Output")
 
-    submit.click(fn=generate_software_spec, 
-                 inputs=[name, description, architecture, components, deployment, platform, extra], 
-                 outputs=output)
+    with gr.Row():
+        with gr.Column():
+            status = gr.Markdown("", visible=False)
+
+    def wrapper(*args):
+        status.update(value="⏳ Generating...", visible=True)
+        result = generate_software_spec(*args)
+        status.update(value="", visible=False)
+        output.update(value=result)
+
+    submit.click(fn=wrapper,
+                 inputs=[name, description, architecture, components, deployment, platform, extra],
+                 outputs=[])
 
 demo.launch()