Fixed issues and resolved conflicts

README.md

@@ -1 +1,6 @@
-# township-small-business-bot
+## API Usage
+
+To test the API locally, ensure your server is running and use the following `curl` command:
+
+```sh
+curl -d '{ "model": "llama3", "prompt": "Why is the sky blue?" }' -H "Content-Type: application/json" http://localhost:11434/api/generate

api_server.py

@@ -0,0 +1,22 @@
+from flask import Flask, request, jsonify
+
+app = Flask(__name__)
+
+@app.route('/api/generate', methods=['POST'])
+def generate_response():
+    data = request.json
+    model = data.get('model')
+    prompt = data.get('prompt')
+    if model and prompt:
+        # Mock response for demonstration
+        response = {
+            "model": model,
+            "prompt": prompt,
+            "response": "The sky is blue due to the scattering of sunlight by the atmosphere."
+        }
+        return jsonify(response), 200
+    else:
+        return jsonify({"error": "Invalid input"}), 400
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=11434)

app.py

@@ -0,0 +1,14 @@
+from flask import Flask, request, jsonify
+
+app = Flask(__name__)
+
+@app.route('/api/generate', methods=['POST'])
+def generate():
+    data = request.json
+    model = data.get('model')
+    prompt = data.get('prompt')
+    response = f"This is a mock response for the model {model} with prompt: {prompt}"
+    return jsonify({"response": response})
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=11434)

ollama_api.py

@@ -0,0 +1,9 @@
+from fastapi import FastAPI
+
+app = FastAPI()
+
+@app.post("/ollama")
+def generate_response(prompt: str):
+    # Replace with your actual Llama 2 model invocation
+    model_response = ollama.generate(prompt, model="llama3", temperature=0.8)
+    return {"response": model_response}