Create app.py

app.py

@@ -0,0 +1,79 @@
+import gradio as gr
+from text_generation import InferenceClient
+
+# ✅ Choose your model (you can change this to another instruct model)
+client = InferenceClient("deepseek-ai/deepseek-coder-6.7b-instruct")
+
+# 🧠 Function to generate software architecture
+def generate_software_spec(name, description, architecture, components, deployment, platform, extra):
+    prompt = f"""
+You are a software architect assistant. Based on the following input, generate:
+1. A **Sequence Diagram** in Mermaid syntax
+2. A **Business Process Flow** in Mermaid syntax
+3. Code Snippets for each component based on the selected tech stack.
+
+---
+
+App Name: {name}
+Description: {description}
+Architecture: {architecture}
+Deployment: {deployment}
+Platform: {platform}
+Components & Tech Stack:
+{components}
+Extra requirements: {extra or "None"}
+
+Return the result in this format:
+
+### Sequence Diagram (Mermaid)
+```mermaid
+<sequence_diagram_here>
+```
+
+### Business Process Flow (Mermaid)
+```mermaid
+<flowchart_here>
+```
+
+### Code Snippets
+#### Component: <Component 1>
+```<language>
+<code_here>
+```
+
+#### Component: <Component 2>
+```<language>
+<code_here>
+```
+"""
+
+    # Generate from model (stream=True recommended for large output)
+    result = ""
+    for chunk in client.text_generation(prompt, stream=True, max_new_tokens=1024, temperature=0.7, stop=["</s>"]):
+        result += chunk.token.text
+    return result
+
+# 🎨 Gradio UI
+with gr.Blocks() as demo:
+    gr.Markdown("## 🧠 AI Software Architecture Assistant")
+
+    with gr.Row():
+        name = gr.Textbox(label="App Name", placeholder="MyApp")
+        description = gr.Textbox(label="Short Description", lines=2, placeholder="A system that manages online tutoring...")
+
+    with gr.Row():
+        architecture = gr.Radio(["Monolithic", "Backend-Frontend"], label="Architecture Style")
+        deployment = gr.Radio(["Serverless", "VM"], label="Deployment Style")
+        platform = gr.Radio(["Web App", "Mobile App"], label="Target Platform")
+
+    components = gr.Textbox(label="Components & Tech Stack (e.g. Backend: Python + Flask, Frontend: React)", lines=4)
+    extra = gr.Textbox(label="Extra Requirements (optional)", lines=2)
+
+    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)
+
+demo.launch()