Update main.py

main.py

@@ -1,59 +1,41 @@
 import asyncio
 import aiohttp
-from fastapi import FastAPI, BackgroundTasks
-from pydantic import BaseModel
-from datetime import datetime
+from fastapi import FastAPI
 import logging
 
-# Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 app = FastAPI()
 
-# The URL of the API we want to call
 API_URL = "https://b917-160-179-178-105.ngrok-free.app/test-ai-call"
 
+call_count = 0
+
 async def call_api():
+    global call_count
     async with aiohttp.ClientSession() as session:
         while True:
             try:
-                async with session.get(API_URL) as response:
+                call_count += 1
+                payload = {"count": call_count}
+
+                async with session.post(API_URL, json=payload) as response:
                     if response.status == 200:
                         data = await response.json()
-                        logger.info(f"API Response: {data}")
+                        logger.info(f"Call {call_count}: API Response: {data}")
                     else:
-                        logger.error(f"API call failed with status code: {response.status}")
+                        logger.error(f"Call {call_count}: API call failed with status code: {response.status}")
+
             except Exception as e:
-                logger.error(f"Error calling API: {e}")
-            
-            # Wait for 10 seconds before the next call
-            await asyncio.sleep(10)
+                logger.error(f"Call {call_count}: Error: {e}")
+
+            await asyncio.sleep(1)
 
 @app.on_event("startup")
 async def startup_event():
-    # Start the background task when the app starts
     asyncio.create_task(call_api())
 
 @app.get("/")
 async def root():
-    return {"message": "API caller is running"}
-
-# You can keep your existing endpoints if needed
-# For example:
-
-class Item(BaseModel):
-    prompt: str
-    history: list
-    system_prompt: str
-    temperature: float = 0.0
-    max_new_tokens: int = 1048
-    top_p: float = 0.15
-    repetition_penalty: float = 1.0
-
-@app.post("/generate/")
-async def generate_text(item: Item):
-    # Your existing generate_text logic here
-    return {"message": "Text generation endpoint"}
-
-# Add any other endpoints or functionality you need
+    return {"message": "API caller is running", "calls_made": call_count}