Update app.py

app.py

@@ -1,11 +1,8 @@
-from flask import Flask, request, jsonify
 import gradio as gr
 from huggingface_hub import InferenceClient
 import json
 
-app = Flask(__name__)
-
-# Inisialisasi Gradio dan HuggingFace client
+# Inisialisasi HuggingFace client
 client = InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct")
 
 def chat_llama(chat_history):
@@ -48,8 +45,7 @@ def process_json(json_input):
     return json.dumps(chat_history, indent=2), ""
 
 # Definisikan antarmuka Gradio
-gr_interface = gr.Blocks()
-with gr_interface:
+with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             chatbot = gr.Chatbot()
@@ -66,31 +62,5 @@ with gr_interface:
             process_btn = gr.Button("Process JSON", variant='primary')
             process_btn.click(fn=process_json, inputs=json_input, outputs=[json_output])
 
-@app.route("/", methods=["GET"])
-def home():
-    return gr_interface.launch(inline=True, share=False)
-
-@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 jsonify(response)
-
-@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 jsonify(response)
-
-@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 jsonify(response)
-
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+# Jalankan antarmuka Gradio dan sediakan API
+demo.launch(server_name="0.0.0.0", server_port=7860, share=False)