commit 0000001

app.py

@@ -0,0 +1,99 @@
+from flask import Flask, request, jsonify, Response
+from flasgger import Swagger, swag_from
+import os
+import requests
+from dotenv import load_dotenv
+
+load_dotenv()
+
+app = Flask(__name__)
+swagger = Swagger(app)
+
+HF_TOKEN = os.environ.get("CHAT_MATE")
+
+@app.route("/chat-stream", methods=["POST"])
+@swag_from({
+    'tags': ['Chat'],
+    'consumes': ['application/json'],
+    'summary': 'Stream assistant reply',
+    'description': 'Send a message and optional history, receive streamed text response.',
+    'parameters': [{
+        'name': 'body',
+        'in': 'body',
+        'required': True,
+        'schema': {
+            'type': 'object',
+            'properties': {
+                'message': {'type': 'string'},
+                'history': {
+                    'type': 'array',
+                    'items': {
+                        'type': 'object',
+                        'properties': {
+                            'role': {'type': 'string'},
+                            'content': {'type': 'string'}
+                        }
+                    }
+                }
+            },
+            'required': ['message']
+        }
+    }],
+    'responses': {
+        200: {
+            'description': 'Streamed reply',
+            'content': {'text/plain': {}}
+        }
+    }
+})
+def chat_stream():
+    data = request.get_json()
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+    response = requests.post(
+        "https://fredericksundeep-aichatmatedev.hf.space/chat-stream",
+        json=data,
+        headers=headers,
+        stream=True
+    )
+    return Response(response.iter_lines(decode_unicode=True), mimetype="text/plain")
+
+@app.route("/chat-stream-doc", methods=["POST"])
+@swag_from({
+    'tags': ['Chat'],
+    'consumes': ['multipart/form-data'],
+    'summary': 'Upload requirement doc & generate code ZIP',
+    'description': 'Upload PDF/TXT with tech preferences. Returns ZIP file.',
+    'parameters': [
+        {'name': 'file', 'in': 'formData', 'type': 'file', 'required': True},
+        {'name': 'frontend', 'in': 'formData', 'type': 'string', 'required': True},
+        {'name': 'backend', 'in': 'formData', 'type': 'string', 'required': True},
+        {'name': 'database', 'in': 'formData', 'type': 'string', 'required': True}
+    ],
+    'responses': {
+        200: {
+            'description': 'ZIP file',
+            'content': {'application/zip': {}}
+        }
+    }
+})
+def chat_stream_doc():
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+    form_data = {
+        "frontend": request.form.get("frontend"),
+        "backend": request.form.get("backend"),
+        "database": request.form.get("database")
+    }
+    files = {'file': request.files['file']}
+    response = requests.post(
+        "https://fredericksundeep-aimateapis.hf.space/chat-stream-doc",
+        headers=headers,
+        data=form_data,
+        files=files
+    )
+    return (response.content, response.status_code, {
+        "Content-Type": response.headers.get("Content-Type", "application/octet-stream"),
+        "Content-Disposition": response.headers.get("Content-Disposition", "attachment; filename=project.zip")
+    })
+
+if __name__ == "__main__":
+    app.run(debug=True)

requirements.txt

@@ -0,0 +1,4 @@
+flask
+flasgger
+requests
+python-dotenv