Update app.py

app.py

@@ -1,23 +1,11 @@
-from fastapi import FastAPI, Request
-from fastapi.responses import HTMLResponse
-from fastapi.middleware.cors import CORSMiddleware
-from huggingface_hub import InferenceClient
+from flask import Flask, request, jsonify
 import gradio as gr
-import uvicorn
-
-# Inisialisasi FastAPI
-app = FastAPI()
+from huggingface_hub import InferenceClient
+import json
 
-# Izinkan CORS (jika perlu)
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
+app = Flask(__name__)
 
-# Inisialisasi Gradio
+# Inisialisasi Gradio dan HuggingFace client
 client = InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct")
 
 def chat_llama(chat_history):
@@ -60,52 +48,49 @@ def process_json(json_input):
     return json.dumps(chat_history, indent=2), ""
 
 # Definisikan antarmuka Gradio
-def gradio_interface():
-    with gr.Blocks() as demo:
-        with gr.Row():
-            with gr.Column():
-                chatbot = gr.Chatbot()
-                msg = gr.Textbox(interactive=True)
-                with gr.Row():
-                    clear = gr.ClearButton([msg, chatbot])
-                    send_btn = gr.Button("Send", variant='primary')
-                msg.submit(fn=chat_mem, inputs=[msg, chatbot], outputs=[msg, chatbot])
-                send_btn.click(fn=chat_mem, inputs=[msg, chatbot], outputs=[msg, chatbot])
-            
-            with gr.Column():
-                json_input = gr.Textbox(placeholder='Input JSON here...', interactive=True, lines=10)
-                json_output = gr.Textbox(label='Output JSON', interactive=False, lines=10)
-                process_btn = gr.Button("Process JSON", variant='primary')
-                process_btn.click(fn=process_json, inputs=json_input, outputs=[json_output])
-    
-    return demo
+gr_interface = gr.Blocks()
+with gr_interface:
+    with gr.Row():
+        with gr.Column():
+            chatbot = gr.Chatbot()
+            msg = gr.Textbox(interactive=True)
+            with gr.Row():
+                clear = gr.ClearButton([msg, chatbot])
+                send_btn = gr.Button("Send", variant='primary')
+            msg.submit(fn=chat_mem, inputs=[msg, chatbot], outputs=[msg, chatbot])
+            send_btn.click(fn=chat_mem, inputs=[msg, chatbot], outputs=[msg, chatbot])
+        
+        with gr.Column():
+            json_input = gr.Textbox(placeholder='Input JSON here...', interactive=True, lines=10)
+            json_output = gr.Textbox(label='Output JSON', interactive=False, lines=10)
+            process_btn = gr.Button("Process JSON", variant='primary')
+            process_btn.click(fn=process_json, inputs=json_input, outputs=[json_output])
 
-@app.get("/", response_class=HTMLResponse)
-async def read_root():
-    return gr.Interface(fn=process_json, inputs="text", outputs="text").launch(inline=True, share=False)
+@app.route("/", methods=["GET"])
+def home():
+    return gr_interface.launch(inline=True, share=False)
 
-@app.post("/chat_llama")
-async def chat_llama_endpoint(request: Request):
-    data = await request.json()
+@app.route("/chat_llama", methods=["POST"])
+def chat_llama_endpoint():
+    data = request.json
     chat_history = data.get('chat_history', [])
     response = chat_llama(chat_history)
-    return response
+    return jsonify(response)
 
-@app.post("/chat_mem")
-async def chat_mem_endpoint(request: Request):
-    data = await request.json()
+@app.route("/chat_mem", methods=["POST"])
+def chat_mem_endpoint():
+    data = request.json
     message = data.get('message', '')
     chat_history = data.get('chat_history', [])
     response = chat_mem(message, chat_history)
-    return response
+    return jsonify(response)
 
-@app.post("/process_json")
-async def process_json_endpoint(request: Request):
-    data = await request.json()
+@app.route("/process_json", methods=["POST"])
+def process_json_endpoint():
+    data = request.json
     json_input = data.get('json_input', '')
     response = process_json(json_input)
-    return response
+    return jsonify(response)
 
-# Jalankan server FastAPI
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    app.run(host="0.0.0.0", port=7860)