Upload groq_flask.py

groq_flask.py

@@ -0,0 +1,50 @@
+from flask import Flask, request, jsonify
+import groq_LPU_inference
+
+app = Flask(__name__)
+
+@app.route('/chat', methods=['GET'])
+def chat():
+    
+    query = request.args.get('query')  # Assuming the query is sent in JSON format
+    model = str(request.args.get('model', "mistral"))  # Optional parameter with default value
+    max_tokens = int(request.args.get('max_tokens', 300))
+    temperature = float(request.args.get('temperature', 0.7))  # Optional parameter with default value
+    assistant = str(request.args.get('assistant', ""))  # Optional parameter with default value
+    system = str(request.args.get('system', "Be Helpful and Friendly. Keep your response straightfoward, short and concise"))  # Optional parameter with default value
+
+    # Developer information
+    developer_info = {
+        'developer': 'Devs Do Code',
+        'contact': {
+            'Telegram': 'https://t.me/devsdocode',
+            'YouTube Channel': 'https://www.youtube.com/@DevsDoCode',
+            'Discord Server': 'https://discord.gg/ehwfVtsAts',
+            'Instagram': {
+                'Personal': 'https://www.instagram.com/sree.shades_/',
+                'Channel': 'https://www.instagram.com/devsdocode_/'
+            }
+        }
+    }
+
+    if query:    
+        response = groq_LPU_inference.Groq_Inference(query, model=model, system=system, assistant=assistant, temp=temperature, max_tokens=max_tokens)
+        return jsonify([{'response': response}, {'developer_info': developer_info}])
+
+    else:
+        error_message = {
+        'developer_contact': {
+            'telegram': 'https://t.me/DevsDoCode',
+            'instagram': 'https://www.instagram.com/sree.shades_/',
+            'discord': 'https://discord.gg/ehwfVtsAts',
+            'linkedin': 'https://www.linkedin.com/in/developer-sreejan/',
+            'twitter': 'https://twitter.com/Anand_Sreejan'
+        },
+        'error': 'Oops! Something went wrong. Please contact the developer Devs Do Code.'
+    }
+        return jsonify(error_message), 400
+
+if __name__ == '__main__':
+    app.run(debug=True)
+
+